#ifndef QCUSTOMTREEVIEW_H
#define QCUSTOMTREEVIEW_H

#include <QTreeView>
#include "mainwindow.h"

class QCustomTreeView : public QTreeView
{
public:
    QCustomTreeView(MainWindow *window);

private:
    void contextMenuEvent(QContextMenuEvent *event);
    MainWindow * window;

};

#endif // QCUSTOMTREEVIEW_H
